
import streamlit as st
import data.data as dd
import time
#设置页面的标签也的名字和icon
st.set_page_config(
    page_title='私人助手登录页面',
    page_icon='🎶'
)
st.title('私人助手登录页面')
username = st.text_input('请输入用户名')
password = st.text_input('请输入密码',type='password')
login_flag = st.button("登录")

def login(username, password):
    if username and password:
        result = dd.query_user_by_username(username)
        if result is None:
            st.error('请检查用户名是否书写正确!')
        else:
            if result["password"] == password:
                st.success("登录成功!")
                time.sleep(2)
                st.session_state.user_id = result["user_id"]
                st.session_state.username = result["username"]
                st.switch_page("pages/chatbot.py")
            else:
                st.error('密码错误,请重新输入!')
    else:
        st.error('信息填写不全!')
#先检验数据是否为空，根据用户名和密码校验数据是否存在
#校验的时候只需要根据用户名查询即可，如果用户名存，再根据获取回来的密码和传入的密码做比对 登录成功之后显示登录成功，延迟两秒到首页(AIz助手聊天界面)
if login_flag:
    login(username, password)



#注册按钮
register_flag = st.button("没有账号？点击注册")
#注册按钮点击之后应该跳转到注册界面
if register_flag:
    #streamlit中有一个函数为switch_page（”py页面文件的路径“）py文件必须位于项目的pages目录下
    st.switch_page("pages/register.py")